package org.usmile.algorithms.leetcode.simple;

import org.usmile.algorithms.leetcode.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * 501. 二叉搜索树中的众数
 *
 * 给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。
 * 如果树中有不止一个众数，可以按 任意顺序 返回。
 * 假定 BST 满足如下定义：
 * 结点左子树中所含节点的值 小于等于 当前节点的值
 * 结点右子树中所含节点的值 大于等于 当前节点的值
 * 左子树和右子树都是二叉搜索树
 *
 * 示例 1：
 * 输入：root = [1,null,2,2]
 * 输出：[2]
 *
 * 示例 2：
 * 输入：root = [0]
 * 输出：[0]
 *
 * 提示：
 * 树中节点的数目在范围 [1, 104] 内
 * -105 <= Node.val <= 105
 *
 * 进阶：你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）
 */
public class _0501 {
}

class _0501_Solution {
    private int prevNum = 0;
    private int count = 0;
    private int maxCount = 0;
    private final List<Integer> result = new ArrayList<>();

    // 中序遍历
    public int[] findMode(TreeNode root) {
        dfs(root);

        int[] res = new int[result.size()];
        for (int i = 0; i < result.size(); i++) {
            res[i] = result.get(i);
        }

        return res;
    }

    private void dfs(TreeNode node) {
        if (node == null) return;
        dfs(node.left);
        update(node.val);
        dfs(node.right);
    }

    private void update(int val) {
        if (val == prevNum) {
            count++;
        } else {
            prevNum = val;
            count = 1;
        }

        if (count == maxCount) {
            result.add(val);
        } else if (count > maxCount) {
            result.clear();
            result.add(val);
            maxCount = count;
        }
    }
}
